Python code for electricity bill calculation